var createError = require('http-errors');
var express = require('express');
var path = require('path');
var { exec } = require('child_process'); // 引入子进程模块
var cookieParser = require('cookie-parser');
var logger = require('morgan');
var cors = require('cors');
var rateLimit = require('express-rate-limit'); // 添加限流中间件
// 这里配置你的路由
var Sms = require('./routes/sms');
var Sen = require('./routes/sne');

const fileUpload = require('express-fileupload');
var app = express();

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'jade');

app.use(fileUpload());
app.use(cors());
app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));


const limiter = rateLimit({
  windowMs: 60 * 1000,
  max: 3000,
  message: '请求过于频繁，请稍后再试。',
  handler: function(req, res) {
    const ip = req.ip; // 获取 IP 地址
    blockIP(ip); // 调屏蔽
    // 设定300解
    setTimeout(() => {
      unblockIP(ip); // 解
    }, 10000); // 10秒
    res.status(429).send('请求过于频繁，请稍后再试。');
  }
});

// 应用限流中间件到所有路由
app.use(limiter);

// 记得写这
app.use('/sne', Sen);
app.use('/sms', Sms);

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  next(createError(404));
});

// error handler
app.use(function(err, req, res, next) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});

module.exports = app;

// 定义函数来执行 iptables 命令来屏蔽 IP 地址
function blockIP(ip) {
  // 执行 iptables 命令
  exec(`sudo iptables -A INPUT -s ${ip} -j DROP`, (err, stdout, stderr) => {
    if (err) {
      console.error('阻止 IP 错误地址:', err);
      return;
    }
    console.log(`被屏蔽的 IP 地址: ${ip}`);
  });
}

// 定义函数来执行 iptables 命令来解除对 IP 地址的屏蔽
function unblockIP(ip) {
  // 执行 iptables 命令
  exec(`sudo iptables -D INPUT -s ${ip} -j DROP`, (err, stdout, stderr) => {
    if (err) {
      console.error('解锁 IP 错误:', err);
      return;
    }
    console.log(`未屏蔽的 IP 地址: ${ip}`);
  });
}
